CandleStick.java

package org.chartsy.candlestick;

import java.awt.Graphics2D;
import java.awt.Rectangle;
import org.chartsy.main.ChartFrame;
import org.chartsy.main.ChartProperties;
import org.chartsy.main.chart.Chart;
import org.chartsy.main.data.ChartData;
import org.chartsy.main.data.Dataset;
import org.chartsy.main.utils.CoordCalc;
import org.chartsy.main.utils.Range;
import org.chartsy.main.utils.SerialVersion;

/**
 *
 * @author viorel.gheba
 */
public class CandleStick
        extends Chart
{

    private static final long serialVersionUID = SerialVersion.APPVERSION;

    public CandleStick()
    {}

    public String getName() 
    { return "Candle Stick"; }

    public void paint(Graphics2D g, ChartFrame cf)
    {
        ChartData cd = cf.getChartData();
        ChartProperties cp = cf.getChartProperties();
		boolean isLog = cp.getAxisLogarithmicFlag();
        Rectangle rect = cf.getSplitPanel().getChartPanel().getBounds();
        rect.grow(-2, -2);
        Range range = cf.getSplitPanel().getChartPanel().getRange();

        if (!cd.isVisibleNull())
        {
            Dataset dataset = cd.getVisible();
            for(int i = 0; i < dataset.getItemsCount(); i++)
            {
                double open = dataset.getOpenAt(i);
                double close = dataset.getCloseAt(i);
                double high = dataset.getHighAt(i);
                double low = dataset.getLowAt(i);

                double x = cd.getX(i, rect);
                double yOpen = cd.getY(open, rect, range, isLog);
                double yClose = cd.getY(close, rect, range, isLog);
                double yHigh = cd.getY(high, rect, range, isLog);
                double yLow = cd.getY(low, rect, range, isLog);
                
                double candleWidth = cp.getBarWidth();
                double candleHeight = Math.abs(yOpen - yClose);
                
                if (open > close ? cp.getBarDownVisibility() : cp.getBarUpVisibility())
                {
                    g.setPaint(open > close ? cp.getBarDownColor() : cp.getBarUpColor());
                    g.fill(CoordCalc.rectangle(x - candleWidth/2, (open > close ? yOpen : yClose), candleWidth, candleHeight));
                }

                if (cp.getBarVisibility())
                {
                    g.setPaint(cp.getBarColor());
                    g.setStroke(cp.getBarStroke());
                    g.draw(CoordCalc.line(x, (open > close ? yOpen : yClose), x, yHigh));
                    g.draw(CoordCalc.line(x, (open > close ? yClose : yOpen), x, yLow));
                    g.draw(CoordCalc.rectangle(x - candleWidth/2, (open > close ? yOpen : yClose), candleWidth, candleHeight));
                }
            }
        }
    }

}